package leo.mystudy.leetcode;

import java.util.List;
import java.util.Stack;

/**
 * @author leo
 * @version 1.0.0
 * @description 给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。
 * <p>
 * k 是一个正整数，它的值小于或等于链表的长度。
 * <p>
 * 如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
 * <p>
 * 进阶：
 * <p>
 * 你可以设计一个只使用常数额外空间的算法来解决此问题吗？
 * 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。
 *  
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：head = [1,2,3,4,5], k = 2
 * 输出：[2,1,4,3,5]
 * 示例 2：
 * <p>
 * <p>
 * 输入：head = [1,2,3,4,5], k = 3
 * 输出：[3,2,1,4,5]
 * 示例 3：
 * <p>
 * 输入：head = [1,2,3,4,5], k = 1
 * 输出：[1,2,3,4,5]
 * 示例 4：
 * <p>
 * 输入：head = [1], k = 1
 * 输出：[1]
 * 提示：
 * <p>
 * 列表中节点的数量在范围 sz 内
 * 1 <= sz <= 5000
 * 0 <= Node.val <= 1000
 * 1 <= k <= sz
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @create 2022/3/17 11:36
 */
public class No25 {

    public static void main(String[] args) {
        ListNode listNode = ListNode.build(new int[]{1, 2, 3, 4, 5});
        System.out.println(new No25().reverseKGroup(listNode, 3));
    }

    private static ListNode tempLast = null;

    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode newHead = null;
        ListNode cur = head;
        Stack<ListNode> stack = new Stack<>();
        while (cur != null) {
            ListNode next = cur.next;
            stack.push(cur);
            if (stack.size() == k) {
                ListNode node = doReverse(stack, next);
                if (newHead == null) {
                    newHead = node;
                }
            }
            cur = next;
        }
//        if (!stack.isEmpty()) {
//            doReverse(stack, null);
//        }
        return newHead;
    }

    private ListNode doReverse(Stack<ListNode> stack, ListNode next) {
        ListNode head = stack.pop();
        if (tempLast != null) {
            tempLast.next = head;
        }
        ListNode cur = head;
        while (!stack.isEmpty()) {
            cur.next = stack.pop();
            cur = cur.next;
        }
        tempLast = cur;
        cur.next = next;
        return head;
    }
}
